import numpy as np
import re

re_complex = re.compile(r'([+-]?[0-9\.]+)([+-][0-9\.]+)i')

def loadTx():
    return np.loadtxt('../data_Tx.txt').shape

def loadY():
    return np.loadtxt('../data_y.txt').shape

def loadLabel():
    return np.loadtxt('../data_test_label.txt').shape

def loadTest():
    with open('../data_test.txt') as f:
        ff = f.readline().strip().split('\t')
        return map(parseComplex,ff)

def parseComplex(s):
    return re_complex.match(s).groups()




if __name__ == '__main__':
    # loadTx()
    # loadY()
    # loadLabel()
    loadTest()
